package tree.tree2.symmetry.ModifyAndConstruct;

import tree.TreeNode;

import java.util.LinkedList;
import java.util.Queue;

/**
 * @ClassName MergeTree
 * @Description TODO
 * @Author lenovo
 * @Date 2023-05-09 10:08
 * @Version 1.0
 * @Comment Magic. Do not touch.
 * If this comment is removed. the program will blow up
 */
public class MergeTree {
    /**
     * 617. 合并二叉树
     * 给你两棵二叉树： root1 和 root2 。
     * <p>
     * 想象一下，当你将其中一棵覆盖到另一棵之上时，两棵树上的一些节点将会重叠（而另一些不会）。你需要将这两棵树合并成一棵新二叉树。合并的规则是：如果两个节点重叠，那么将这两个节点的值相加作为合并后节点的新值；否则，不为 null 的节点将直接作为新二叉树的节点。
     * <p>
     * 返回合并后的二叉树。
     * <p>
     * 注意: 合并过程必须从两个树的根节点开始。
     */

    public TreeNode mergeTrees(TreeNode root1, TreeNode root2) {
        if (root1 == null) return root2;
        if (root2 == null) return root1;

        Queue<TreeNode> queue = new LinkedList<>();
        queue.add(root1);
        queue.add(root2);
        while (!queue.isEmpty()) {
            TreeNode poll1 = queue.poll();
            TreeNode poll2 = queue.poll();
            poll1.val = poll1.val + poll2.val;

            if (poll1.left != null && poll2.left != null) {
                queue.add(poll1.left);
                queue.add(poll2.left);
            }

            if (poll1.right != null && poll2.right != null) {
                queue.add(poll1.right);
                queue.add(poll2.right);
            }
            if (poll1.left == null && poll2.left != null) {
                poll1.left = poll2.left;
            }

            if (poll1.right == null && poll2.right != null) {
                poll1.right = poll2.right;
            }

        }
        return root1;
    }

    public static void main(String[] args) {
        TreeNode root1 = new TreeNode(1);
        root1.left = new TreeNode(2);
        root1.right = new TreeNode(2);
        TreeNode root2 = new TreeNode(1);
        root2.right = new TreeNode(2);
        MergeTree mergeTree = new MergeTree();
        System.out.println(mergeTree.mergeTrees(root1, root2));
    }
}